import numpy as np
import matplotlib.pyplot as plt
img = np.random.rand(10, 10)
plt.imshow(img, cmap = "gray")
plt.colorbar()
plt.show()
plt.imsave(r"D:\l'j't\code\pycharm\learning-python\AI\images\test_save.png", img)